import pytest
import redis

from connect_data_src import ConnectDataSource


class ConnectionRedis(ConnectDataSource):
    def __init__(self):
        self.redis = ConnectDataSource().connect_redis()
        self.datasrc = ConnectDataSource().read_database_conf()
        # 数据文件中的客户id列表
        self.customer_ids = self.datasrc["customer_ids"]

    def hset_value(self):
        # 实现插入hash数据
        # print(r.get("customer_id"))
        self.redis.hset("1000000002", "login_time", "2021-08-21")
        self.redis.hset("1000000002", "mobie", "Android")
        return self.redis.exists("1000000002")

    def hdelete_value(self):
        # todo 这里只能传递一个id，后续继续完善
        self.redis.delete(self.customer_ids[0])
        # return self.redis.hgetall(self.customer_ids[0])
        # 返回该customerID是否存在的，存在1，不存在0
        return self.redis.exists(self.customer_ids[0])
